package sliding_window

/*
1343.给你一个整数数组arr和两个整数 k和 threshold。
请你返回长度为 k且平均值大于等于threshold的子数组数目。

示例 1：
输入：arr = [2,2,2,2,5,5,5,8], k = 3, threshold = 4
输出：3
解释：子数组 [2,5,5],[5,5,5] 和 [5,5,8] 的平均值分别为 4，5 和 6 。其他长度为 3 的子数组的平均值都小于 4 （threshold 的值)。

示例 2：
输入：arr = [11,13,17,23,29,31,7,5,2,3], k = 3, threshold = 5
输出：6
解释：前 6 个长度为 3 的子数组平均值都大于 5 。注意平均值不是整数。


提示：
	1 <= arr.length <= 105
	1 <= arr[i] <= 104
	1 <= k <= arr.length
	0 <= threshold <= 104

*/

//滑动窗口模板
func numOfSubArrays(arr []int, k int, threshold int) int {
	right, left, sum, ans := 0, 0, 0, 0

	//expand windows
	for right < len(arr) {
		sum += arr[right]

		//shrink windows
		for right-left+1 >= k {
			// 长度为k
			if right-left+1 == k {
				avg := float64(sum) / float64(k)
				//平均值大于等于threshold
				if avg >= float64(threshold) {
					ans++
				}

				sum -= arr[left]
				left++
			}
		}
		right++
	}
	return ans
}

// 滑动窗口+动态规划（区间）
func numOfSubArraysDC(arr []int, k int, threshold int) int {
	sum, result, sumTarget := 0, 0, k*threshold

	for i := 0; i < k; i++ {
		sum += arr[i]
	}

	adder := sum - sumTarget // adder 可能为负值
	if adder >= 0 {
		result++
	}

	pos := k
	for j := 0; j < len(arr)-k; j++ {
		adder = adder + arr[pos] - arr[j] //动态规划去计算区间：adder存储了第一次求和的值 所以更新窗口后 不用重新把所有数加一遍
		if adder >= 0 {
			result++
		}
		pos++
	}
	return result
}
